Skip to content

feat: s3 transfer manager v2 #3079

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 56 commits into
base: master
Choose a base branch
from

Conversation

yenfryherrerafeliz
Copy link
Contributor

@yenfryherrerafeliz yenfryherrerafeliz commented Feb 6, 2025

Description of changes:

  • Transfer Manager V2 implementation.
  • This changes includes:
    • Basic S3 Transfer Manager Client.
    • Multipart Download Functionality
    • Progress Tracker interfaces, along with a default console progress tracker.

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

This is an initial phase for the s3 transfer manager v2, which includes:
- Progress Tracker with a default Console Progres Bar.
- Dedicated Multipart Download Listener for listen to events specificly to multipart download.
- Generic Transfer Listener that will be used in either a multipart upload or a multipart download. The progress tracker is dependant on the Generic Transfer Listener, and when enabled it uses the same parameter to be provided as the progress tracker. This is important because if there is a need for listening to transfer specific events and also track the progress then, a custom implementation must be done that incorporate those two needs together, otherwise one of each other must be used.
- Single Object Download
- Multipart Objet Download

This initial implementation misses the test cases.
- Refactor set a single argument, even when not exists, in the console progress bar.
- Add a specific parameter for showing the progress rendering defaulted to STDOUT.
- Add test cases for ConsoleProgressBar.
- Add test cases for DefaultProgressTracker.
- Add test cases for ObjectProgressTracker.
- Add test cases for TransferListener.
- Add test cases for multipart download listener.
- Add a trait to the MultipartDownloader implementation to keep the main implementatio cleaner.
- Add test cases for multipart downloader, in specific testing part and range get multipart downloader.
Copy link
Member

@stobrien89 stobrien89 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looking good on the first pass- there are some nits like function braces needing newlines, new files needing newlines, naming conventions, etc. also had some questions about design

Refactor:
- Moves opening braces into a new line.
- Make requestArgs an optional argument.
- Remove unnecessary traits.
- Use traditional declarations.

Adds:
- Download directory feature.
Refactor:
- Add a message placeholder for progress status. For example in case of errors.

Adds:
- Upload feature, missing multipart functionality.
- Add upload directory feature
- Add a dedicated multipart upload implementation
- Add transfer progress to multipart upload
- Add upload directory with the required options.
- Create specific response models for upload, and upload directory.
- Add multipart upload test cases.
- Fix transfer listener completation eval.
Short namespace from `Aws\S3\Features\S3Transfer` to `Aws\S3\S3Transfer`.
Copy link
Member

@stobrien89 stobrien89 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

A few more comments- I think a few from the last round were left addressed also. I'd do another check for function opening braces (needing to be moved to a new line) and new files that are missing a newline at the end. More test classes needed as well but I'm assuming those are on the way

- Implement progress tracker based on SEP spec.
- Add a default progress bar implementation.
- Add different progress tracker formats:
-- Plain progress format: [|progress_bar|] |percent|%
-- Transfer progress format: [|progress_bar|] |percent|% |transferred|/|tobe_transferred| |unit|
-- Colored progress format: |object_name|:\n\033|color_code|[|progress_bar|] |percent|% |transferred|/|tobe_transferred| |unit| |message|\033[0m
- Add a default single progress tracker implementation.
- Add a default multi progress tracker implementation for tracking directory transfers.
- Include tests unit just for console progress bar.
- Fixes current test cases for:
  - MultipartUploader
  - MultipartDownloader
  - ProgressTracker
- Remove progress bar color enum since the colors were moved into the specific format that requires them.
TransferListener must be tested from the implementations that extends and use this abstract class.
Add nullable type to listenerNotifier property in the MultipartUploader implementation.
- Tests for MultiProgressTracker
- Tests for SingleProgressTracker
- Tests for ProgressBarFormat
- Tests for TransferProgressSnapshot
- Tests for TransferListenerNotifier
Copy link
Member

@stobrien89 stobrien89 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looking better- still needing some more unit tests, along with integ tests. Left some comments and nits on formatting. It seems each new file is missing a newline so I'd check those as well

- Refactor code to address some styling related feedback.
- Add upload and uploadDirectory unit tests.
- Fix MultipartUpload tests by increasing the part size from 1024 to 10240000 so it gets between the allowed part size range 5MB-5GBs.
- Rename tobe to to_be in the progress formatting.
- Add download tests
- Add download directory tests
- Minor naming refactor
Copy link
Member

@stobrien89 stobrien89 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looking good. Just a few nits this time around. Still needing integ tests- will do another round of reviews once those are up.

- Add upload integ tests for:
 - Single uploads
 - Multipart uploads
 - Checksum in single uploads
 - Checksum in multipart uploads
- Add download integ tests for:
 - Single downloads
 - Multipart downloads
- Add integ tests for directory uploads
- Add integ tests for directory downloads
Copy link
Member

@stobrien89 stobrien89 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Mostly nits, but the most important requests are: adding upload() and download() methods on the uploader and downloader classes that call promise() (similar to the old implementation)and testing the resolvesOutsideTargetDirectory logic.

I would do an audit of hard-coded values that can be moved to classes, line length (max 85 char) and all new files that do not end with a newline.

- Move some fixed values out of the methods into consts.
- Address a line exceeded 80 chars.
- Declare keys used across different implementations as consts.
- Fix keys declaration in TransferListener.php
- Make use of DIRECTORY_SEPARATOR const instead of hardcoding `/`
- Some implementations using TransferListener were missing the import statement.
- Removed old MultipartUploader and DownloaderImplementation which were suffixed with [class]Initial.php
- Remove empty space.
When a s3 prefix is provided and the object key contains the delimiter then, the prefix should be stripped off from the object key.
- when the directory separator is different from the s3 delimiter then the separator from the object key is replaced with the os directory separator.
- Update the tests to validate object key destination correctly whe using the download directory API.
The delimiter for the list object request in a download directory must be defaulted to null unless explicitly provided in the list object v2 args.
The data provider for testResolvesOutsideTargetDirectory was renamed to `resolvesOutsideTargetDirectoryProvider`.
private StreamInterface | string $source;

/** @var array */
private array $putObjectRequestArgs;
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Related to my other comment in MultipartUploader: should we have this live on S3TransferManager? that's the only place we're using the putObject operation

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

And if so, do we want something like $createMultipartUploadArgs, $uploadPartArgs and $completeMultipartUploadArgs for the multipart workflows?

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Same comment as the one I made for the same param above- putObjectRequestArgs implies it will only be used for putObject requests, which are only performed for single uploads. From what I can see, this parameter is also used for Multipart uploads

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Would it also be possible to rename to UploadDirectoryResult extend Aws\Result here with some overrides (i.e.) keep some of this logic in place?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

What would be the use case for UploadDirectoryResult to extend Aws\Result? Right now this class just has two counting properties, one for successful uploads and one for failed uploads. Like:

class UploadDirectoryResult {
      int $failedUploads;
      int $successfulUploads;
}

In AbstractMultipartUploader the parameter used to be putObjectRequestArgs, but it makes more sense to name it requestArgs.
- Make config arguments optional
- Make getObjectRequest/putObjectRequest argument optional
- Delete file if exists when using the file download handler.
- Make getObjectRequestArgs optional in download directory operation.
- Add config key in docs
Copy link
Member

@stobrien89 stobrien89 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Overall, this is looking great. Thanks for dealing with all of the churn. Mostly formatting comments and a few exception handling suggestions

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I would make sure to check all new files for missing newlines at the end

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Still seeing a few of these


namespace Aws\S3\S3Transfer\Models;

class DownloadDirectoryResponse
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

can we change this to DownloadDirectoryResult and extend Aws\Result?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I am not sure what would be the use case right now for doing this. This class just holds how many uploads failed and how many succeeded. So it is not hold any other data.

Copy link
Member

@stobrien89 stobrien89 Jul 25, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Got it. I think the name change should still apply. Is there a way to expose (a group of) the individual DownloadResult objects here?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Yes, I think we should expose the download results here. Or at least for failed request. I will think what could be the best approach here, because, I don't want to have a class holding so many results which could cause unnecessary memory consumption.

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Agreed. Probably not ideal for successful downloads. But we should probably include all of the results for successful uploads on its corresponding class

Copy link
Member

@stobrien89 stobrien89 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Some of the prior comments I "un-resolved" may be related to the new diff view I've been using. let me know if you're seeing any discrepancies

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Still seeing a few of these

*
* @return PromiseInterface
*/
public function download(
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Probably the wrong entry point for the comment, but I saw we had recursive directory upload functionality. We should have that for downloads as well per the spec, but I don't think that's a hard requirement

- Make data model final
- Refactor test cases to use correct parameters for S3TransferManager APIs
- Fix S3TransferManagetContext.php to use the correct parameter expected in the different APIs exposed by S3TranserManager
- Refactor some formatting styling.
- Added empty line at the end of files.
- Add a more descriptive documentation in $failsWhenDestinationExists.
- Remove unnecessary break line.
];

/** @var StreamInterface|string */
private StreamInterface | string $source;
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

this one still needs to be addressed

* @return StreamInterface
*/
private function parseBody(
string | StreamInterface $source
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This one still needs to be addressed

- Remove spaces between union type definitions
- Add documentation for config parameters in UploadDirectoryRequest.
- Add missing new lines in a few places.
- Rename upload and download request args.
- Refactor tests to clean up resources specifically in the finally block instead of using cleanUpFns.
- Add Throwable and Result type.
- Make parameter optional by adding a default value.
- Source should have been provided as null when the bucket and key are already provided as part of the download request args.
- When checksum type will be resolved to FULL_OBJECT the only supported algorithms are CRC family.
Copy link
Member

@stobrien89 stobrien89 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Getting close! Another couple of comments

protected ?TransferProgressSnapshot $currentSnapshot;

/**
* This will be used for custom or default checksum.
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

nit: can you remove "This will be used for" and keep the rest?

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I would make sure all of the test fixtures from the specification are included: There are fixture files for single upload/download and tables for Upload/Download directory cases

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

nit: we should use the Aws\S3\S3Transfer\Exception namespace here and change the dir name to Exception

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This seems like it should be in the Models. But I'm now realizing that I missed namespacing in other reviews. It would make sense to do away with Models and create separate directories and namespaces for Upload, Download, Request and Response. Exception should also be a namespace, but I've commented on that already. What do you think?

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This seems like it should also be in another location. My comment about namespacing applies to test classes as well

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants